from easycore.common.registry import Registry

DATASET_REGISTRY = Registry('dataset')
DATASET_INFO_REGISTRY = Registry("dataset info")

DATASET_OUTPUT_DIR = 'output/dataset'
DATASET_CONFIG_DIR = 'configs/dataset'

def build_dataset(*dataset_name_list):
    dataset_dicts = []
    for dataset_name in dataset_name_list:
        dataset_dicts += DATASET_REGISTRY.get(dataset_name)()
    return dataset_dicts